from .ssd_detector import SSDDetector

# 将支持的目标检测网络与对应的名字保存在一个map中
_DETECTION_META_ARCHITECTURES = {
    "SSDDetector": SSDDetector # 这个就是我们使用的SSD侦测网络，这里是将构造函数作为value存储在dict里面
}


def build_detection_model(cfg):
    '''
    根据配置文件，创建对应的ssd网络
    '''
    meta_arch = _DETECTION_META_ARCHITECTURES[cfg.MODEL.META_ARCHITECTURE]
    # 这个动作就是网络加载配置文档，调用构造函数，构造一个ssd检测网络
    return meta_arch(cfg)
